SelfDR: Self-Distillation from Reasoning for LLM-Based Recommendation
SelfDR:基于推理自蒸馏的 LLM 推荐框架
🧭 LLM 推荐 · 自蒸馏推理增强 ★ 深度解读
Tsinghua University · Quan Cheng Laboratory · Meituan
用推荐性能为 reward 训练 reasoner,再通过动态加权自蒸馏将推理增强迁移到直接推荐模型,在线推理零额外开销
📌 工作参考价值(4 点)
- 在线延迟敏感场景可直接复用:训练时用 GRPO 以推荐命中率为 reward 训练一个 reasoner 生成用户偏好解释,将解释拼入 teacher 输入得到更强排序分布;在线部署 student 直接输出候选 ID,不产生额外推理 token,实测推理时间与普通 SFT 模型相同。
- 动态蒸馏权重公式值得借鉴:根据 teacher 对正样本的排序位置、teacher 与 student 的 rank 差动态调整蒸馏 loss 权重,teacher 排序差时大幅降权,避免从噪声中学习;超参 β≈0.1、温度系数 1~2 较稳。
- 蒸馏损失用 reverse KL 而非 forward KL,让 student 分布集中到 teacher 高概率区域,适合候选 ID 排序场景(候选集有限,需要聚焦 top 候选)。
- 训练 reasoner 时对 rationale 做细粒度 mask(连续 3 个词与 item 标题重叠则 mask),防止信息泄露,提升 rationale 质量;此策略可迁移到任何用生成解释辅助推荐的 pipeline。